parser = argparse.ArgumentParser()
parser.add_argument("--do_train", action="store_true")
parser.add_argument("--do_eval", action="store_true")
parser.add_argument("--do_predict", action="store_true")
parser.add_argument("--model_path",default='/path/to/model',type=str)
parser.add_argument("--save_path", default="lcsts",type=str)
parser.add_argument("--task", default="lcsts",type=str)
parser.add_argument("--gradient_accumulation_steps", default="4",type=str)
parser.add_argument("--lr",default=3e-6,type=float)
parser.add_argument("--batch_size",default='32',type=str)
parser.add_argument("--epoch",default='10',type=str)
# parser.add_argument("--warmup_steps",default='1000',type=str)
parser.add_argument("--lr_scheduler",default='polynomial',type=str)
parser.add_argument("--save_strategy",default='epoch',type=str)
parser.add_argument("--data_dir",default="/path/to/dataset/",type=str)
parser.add_argument("--seed",default='2021',type=str)
parser.add_argument("--predict_file",default='results',type=str)
args = parser.parse_args()
arg_dict=args.__dict__